package com.hbk.programe;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

public class Dto {
	
    private String clazz;
    private List<DtoField> fields=new ArrayList<>();
	public String getClazz() {
		return clazz;
	}
	public void setClazz(String clazz) {
		this.clazz = clazz;
	}
	public List<DtoField> getFields() {
		return fields;
	}
	public void setFields(List<DtoField> fields) {
		this.fields = fields;
	}
	
	public Set<String> getImports(){
		Set<String> list=new HashSet<>();
		for (DtoField field : fields) {
			if(field.getType().equals("BigDecimal")) {
				list.add("import java.math.BigDecimal;");
			}
			if(field.getType().equals("Date")) {
				list.add("import java.util.Date;");
			}
			if(field.getType().contains("List")) {
				list.add("import java.util.List;");
				list.add("import java.util.ArrayList;");
			}
			if(field.getType().equals("Set")) {
				list.add("import java.util.Set;");
				list.add("import java.util.HashSet;");
			}
		}
		return list;
	}
	
}
